Efficient Python loops